package org.example.util;

import org.example.model.User;
import org.example.util.WebUtil;

import javax.servlet.http.HttpSession;
import javax.websocket.HandshakeResponse;
import javax.websocket.server.HandshakeRequest;
import javax.websocket.server.ServerEndpointConfig;

//websocket的配置类：建立连接之前，定义一个配置
public class WebsocketConfigurator extends ServerEndpointConfig.Configurator {

    //进行握手阶段的一些配置：在客户端与服务端建立websocket连接，服务端就可以使用该配置完成初始化

    @Override
    public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
        //websocket建立连接时，执行配置的代码 然后执行onOpen()
        HttpSession httpSession =(HttpSession) request.getHttpSession();
        //保存session
        if (httpSession!=null){
            sec.getUserProperties().put("HttpSession",httpSession);
        }

    }
}
